Amazon Bedrock に追加された Stability AI の画像生成モデル Top3 が、AWS CDK でモデル識別子として利用可能になりました

npm i aws-cdk-lib@latest aws-cdk@latest

import * as bedrock from 'aws-cdk-lib/aws-bedrock';
import * as cdk from 'aws-cdk-lib';
import { Construct } from 'constructs';

export class CdkSampleStack extends cdk.Stack {
  constructor(scope: Construct, id: string) {
    super(scope, id);

    /**
     * Stable Image Ultra モデルの識別子を取得
     */
    const stableImageUltraModel = bedrock.FoundationModel.fromFoundationModelId(
      this,
      'StableImageUltraModel',
      bedrock.FoundationModelIdentifier.STABILITY_STABLE_IMAGE_ULTRA_V1_0
    );

    // モデルの識別子を出力
    new cdk.CfnOutput(this, 'StableImageUltraModel', {
      value: stableImageUltraModel.modelId,
    });

    /**
     * Stability SD3 Large モデルの識別子を取得
     */
    const stabilitySd3LargeModel =
      bedrock.FoundationModel.fromFoundationModelId(
        this,
        'StabilitySd3LargeModel',
        bedrock.FoundationModelIdentifier.STABILITY_SD3_LARGE_V1_0
      );

    // モデルの識別子を出力
    new cdk.CfnOutput(this, 'StabilitySd3LargeModel', {
      value: stabilitySd3LargeModel.modelId,
    });

    /**
     * Stability Core モデルの識別子を取得
     */
    const stableImageCoreModel = bedrock.FoundationModel.fromFoundationModelId(
      this,
      'StableImageCoreModel',
      bedrock.FoundationModelIdentifier.STABILITY_STABLE_IMAGE_CORE_V1_0
    );

    // モデルの識別子を出力
    new cdk.CfnOutput(this, 'StableImageCoreModel', {
      value: stableImageCoreModel.modelId,
    });
  }
}

$ npm run deploy

> cdk_sample_app@0.1.0 deploy
> cdk deploy --require-approval never --method=direct

✨  Synthesis time: 5.29s

CdkSampleStack:  start: Building 5cb35d407b13473f24bcde3a2a66adf42a211229a54d7c51c5d6a815b192c856:current_account-current_region
CdkSampleStack:  success: Built 5cb35d407b13473f24bcde3a2a66adf42a211229a54d7c51c5d6a815b192c856:current_account-current_region
CdkSampleStack:  start: Publishing 5cb35d407b13473f24bcde3a2a66adf42a211229a54d7c51c5d6a815b192c856:current_account-current_region
CdkSampleStack:  success: Published 5cb35d407b13473f24bcde3a2a66adf42a211229a54d7c51c5d6a815b192c856:current_account-current_region
CdkSampleStack: deploying... [1/1]
CdkSampleStack: updating stack...

 ✅  CdkSampleStack

✨  Deployment time: 161.36s

Outputs:
CdkSampleStack.StabilitySd3LargeModel = stability.sd3-large-v1:0
CdkSampleStack.StableImageCoreModel = stability.stable-image-core-v1:0
CdkSampleStack.StableImageUltraModel = stability.stable-image-ultra-v1:0
Stack ARN:
arn:aws:cloudformation:ap-northeast-1:XXXXXXXXXXXX:stack/CdkSampleStack/fd7023d0-653d-11ef-b8c3-0e28b5cf9e67

import os
import boto3
import json
import base64
import io
from PIL import Image

bedrockModelId = os.environ[BEDROCK_MODEL_ID] # Bedrock の モデル識別子を環境変数から取得

bedrock = boto3.client('bedrock-runtime', region_name='us-west-2')
response = bedrock.invoke_model(
    modelId=bedrockModelId,
    body=json.dumps({
        'prompt': 'A car made out of vegetables.'
    })
)
output_body = json.loads(response["body"].read().decode("utf-8"))
base64_output_image = output_body["images"][0]
image_data = base64.b64decode(base64_output_image)
image = Image.open(io.BytesIO(image_data))
image.save("image.png")

import * as bedrock from 'aws-cdk-lib/aws-bedrock';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as cdk from 'aws-cdk-lib';
import { Construct } from 'constructs';
import { PythonFunction } from '@aws-cdk/aws-lambda-python-alpha';

export class CdkSampleStack extends cdk.Stack {
  constructor(scope: Construct, id: string) {
    super(scope, id);

    /**
     * Stable Image Ultra モデルの識別子を取得
     */
    const stableImageUltraModel = bedrock.FoundationModel.fromFoundationModelId(
      this,
      'StableImageUltraModel',
      bedrock.FoundationModelIdentifier.STABILITY_STABLE_IMAGE_ULTRA_V1_0
    );

    /**
     * 注入された ID のモデルを使用する Lambda 関数を作成
     */
    new PythonFunction(this, 'HelloFunction', {
      runtime: lambda.Runtime.PYTHON_3_12,
      entry: 'src/lambda/hello',
      handler: 'handler',
      environment: {
        BEDROCK_MODEL_ID: stableImageUltraModel.modelId,
      },
    });
  }
}